package org.hzero.system.xml;


import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.util.Assert;
import org.springframework.util.StringUtils;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.Marshaller;
import javax.xml.bind.annotation.XmlRootElement;
import java.io.ByteArrayOutputStream;
import java.util.List;

/**
 * @ClassName: XmlUtil
 * @Description: xml转化为对象
 * @author: tianhao.luo@hand-china.com
 * @date: 2020/11/17  17:56
 */
public final class KnifeXmlUtil {

    private static final Logger LOGGER = LoggerFactory.getLogger(KnifeXmlUtil.class);


    /**
     * 将xml转换为json对象
     */
    public static JSONObject xmlToJson(String xml,String rootName) throws DocumentException {
        JSONObject jsonObject = new JSONObject();
        Document document = DocumentHelper.parseText(xml);
        //获取根节点元素对象
        Element root = document.getRootElement();
        if (!StringUtils.isEmpty(rootName)){
            root = getOrderNode(root,rootName);
        }
        iterateNodes(root, jsonObject);
        return jsonObject;
    }

    /**
     * 获取指定节点
     * @param root 根节点
     * @param rootName 指定节点
     * @return 指定节点的element
     */
    private static Element getOrderNode(Element root,String rootName){
        List<Element> elements = root.elements();
        Element temp = null;
        for (Element element :
                elements) {
            if (element.getName().equals(rootName)) {
                temp =  element;
            }else {
                temp = getOrderNode(element,rootName);
            }
        }
        return temp;
    }

    /**
     * 遍历元素
     * @param node 节点
     * @param json json对象
     */
    private static void iterateNodes(Element node, JSONObject json) {
        //获取当前元素名称
        String nodeName = node.getName();
        //判断已遍历的JSON中是否已经有了该元素的名称
        if(json.containsKey(nodeName)){
            //该元素在同级下有多个，拿出旧的节点信息
            Object obj = json.get(nodeName);
            JSONArray array = null;
            if(obj instanceof JSONArray){
                array = (JSONArray) obj;
            }else {
                array = new JSONArray();
                //将旧节点加入到数组中
                array.add(obj);
            }
            //获取该元素下所有子元素
            List<Element> listElement = node.elements();
            if(listElement.isEmpty()){
                //该元素无子元素，获取元素的值
                String nodeValue = node.getTextTrim();
                array.add(nodeValue);
                json.put(nodeName, array);
                return ;
            }
            //有子元素，将子元素组成一个json对象
            JSONObject newJson = new JSONObject();
            //遍历所有子元素
            for(Element e:listElement){
                //递归
                iterateNodes(e,newJson);
            }
            array.add(newJson);
            //利用json对象是键值对类型的map来替换旧的数组
            json.put(nodeName, array);
            return ;
        }
        //该元素同级下第一次遍历
        //获取该元素下所有子元素
        List<Element> listElement = node.elements();
        if(listElement.isEmpty()){
            //该元素无子元素，获取元素的值
            String nodeValue = node.getTextTrim();
            json.put(nodeName, nodeValue);
            return ;
        }
        //有子节点，新建一个JSONObject来存储该节点下子节点的值
        JSONObject object = new JSONObject();
        for(Element e:listElement){
            //递归
            iterateNodes(e,object);
        }
        json.put(nodeName, object);
        return ;
    }

    /**
     * 获取对象转化成的xml内容
     * @param obj 对象
     * @param <T> 泛型对象
     * @return xml内容
     * @throws Exception 异常信息
     */
    public static <T> String objToXml(T obj) throws Exception{
        Class<?> aClass = obj.getClass();
        XmlRootElement annotation = aClass.getAnnotation(XmlRootElement.class);
        Assert.isTrue(annotation !=null,"请在需要转化为xml的类上添加注解@XmlRootElement(name = \"arg0\")\n" +
                "@XmlAccessorType(value = XmlAccessType.FIELD)");
        String simpleName = annotation.name();
        //构造JAXBContext
        JAXBContext context = JAXBContext.newInstance(aClass);
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        Marshaller marshaller = context.createMarshaller();
        marshaller.marshal(obj,outputStream);
        String xmlContent = outputStream.toString();
        int start = xmlContent.indexOf("<" + simpleName + ">");
        int end = xmlContent.indexOf("</" + simpleName + ">");
        String substring = xmlContent.substring(start, end + simpleName.length() + 3);
        LOGGER.info("对象转化xml内容："+substring);
        return substring;
    }
}
